Room layout estimation methods and techniques

ABSTRACT

Systems and methods for estimating a layout of a room are disclosed. The room layout can comprise the location of a floor, one or more walls, and a ceiling. In one aspect, a neural network can analyze an image of a portion of a room to determine the room layout. The neural network can comprise a convolutional neural network having an encoder sub-network, a decoder sub-network, and a side sub-network. The neural network can determine a three-dimensional room layout using two-dimensional ordered keypoints associated with a room type. The room layout can be used in applications such as augmented or mixed reality, robotics, autonomous indoor navigation, etc.

CROSS-REFERENCE TO RELATED APPLICATIONS

This application is a continuation of U.S. patent application Ser. No. 15/923,511, filed Mar. 16, 2018, entitled “ROOM LAYOUT ESTIMATION METHODS AND TECHNIQUES,” which claims the benefit of priority to U.S. Patent Application No. 62/473,257, filed Mar. 17, 2017, entitled “ROOM LAYOUT ESTIMATION METHODS AND TECHNIQUES,” which is hereby incorporated by reference herein in its entirety.

FIELD

The present disclosure relates generally to systems and methods for estimating a layout of a room using automated image analysis and more particularly to deep machine learning systems (e.g., convolutional neural networks) for determining room layouts.

BACKGROUND

A deep neural network (DNN) is a computational machine learning method. DNNs belong to a class of artificial neural networks (NN). With NNs, a computational graph is constructed which imitates the features of a biological neural network. The biological neural network includes features salient for computation and responsible for many of the capabilities of a biological system that may otherwise be difficult to capture through other methods. In some implementations, such networks are arranged into a sequential layered structure in which connections are unidirectional. For example, outputs of artificial neurons of a particular layer can be connected to inputs of artificial neurons of a subsequent layer. A DNN can be a NN with a large number of layers (e.g., 10s, 100s, or more layers).

Different NNs are different from one another in different perspectives. For example, the topologies or architectures (e.g., the number of layers and how the layers are interconnected) and the weights of different NNs can be different. A weight can be approximately analogous to the synaptic strength of a neural connection in a biological system. Weights affect the strength of effect propagated from one layer to another. The output of an artificial neuron can be a nonlinear function of the weighted sum of its inputs. A NN can be trained on training data and then used to determine an output from untrained data.

SUMMARY

Building a three-dimensional (3D) representation of the world from an image is an important challenge in computer vision and has important applications to augmented reality, robotics, autonomous navigation, etc. The present disclosure provides examples of systems and methods for estimating a layout of a room by analyzing one or more images of the room. The layout can include locations of a floor, one or more walls, a ceiling, and so forth in the room.

In one aspect, a machine learning system comprising a neural network is used for room layout estimation. In various embodiments, the machine learning system is referred to herein by the name RoomNet, because these various embodiments determine a Room layout using a neural Network. The machine learning system can be performed by a hardware computer processor comprising non-transitory storage and can be performed locally or in a distributed (e.g., cloud) computing environment.

The room layout systems and methods described herein are applicable to augmented and mixed reality. For example, an augmented reality (AR) device can include an outward-facing imaging system configured to capture an image of the environment of the AR device. The AR device can perform a RoomNet analysis of the image to determine the layout of a room in which a wearer of the AR device is located. The AR device can use the room layout to build a 3D representation (sometimes referred to as a world map) of the environment of the wearer.

In one aspect, a neural network can analyze an image of a portion of a room to determine the room layout. The neural network can comprise a convolutional neural network having an encoder sub-network, a decoder sub-network, and a side sub-network. The neural network can determine a three-dimensional room layout using two-dimensional ordered keypoints associated with a room type. The room layout can be used in applications such as augmented or mixed reality, robotics, autonomous indoor navigation, etc.

In one aspect, RoomNet comprises an encoder sub-network, a decoder sub-network connected to the encoder network, and a side sub-network connected to the encoder network. After receiving a room image, a plurality of predicted heat maps corresponding to a plurality of room types can be determined using the encoder sub-network and the decoder sub-network of the RoomNet. A predicted room type of the plurality of room types can be determined using the encoder sub-network and the side sub-network of the RoomNet and the room image. Keypoints at a plurality of predicted keypoint locations can be determined using a predicted heat map corresponding to the predicted room type. A predicted layout of a room in the room image can be determined using the predicted room type, the keypoints, and a keypoint order associated with the predicted room type.

In another aspect, a system is used to train a neural network for room layout estimation. Training room images can be used to train the neural network, which can comprise an encoder sub-network, a decoder sub-network connected to the encoder network, and a side sub-network connected to the encoder network. Each of the training room images can be associated with a reference room type and reference keypoints at a reference keypoint locations in the training room image. Training the neural network can include determining, using the encoder sub-network and the decoder sub-network and the training room image, a plurality of predicted heat maps corresponding to the room types, and determining, using the encoder sub-network and the side sub-network and the training room image, a predicted room type. The neural network can include weights that are updated based on a first difference between the reference keypoint locations and a predicted heat map and a second difference between the reference room type and the predicted room type.

Details of one or more implementations of the subject matter described in this specification are set forth in the accompanying drawings and the description below. Other features, aspects, and advantages will become apparent from the description, the drawings, and the claims. Neither this summary nor the following detailed description purports to define or limit the scope of the inventive subject matter.

BRIEF DESCRIPTION OF THE DRAWINGS

FIG. 1A is an example pipeline for room layout estimation.

FIG. 1B is an example room layout estimation using an embodiment of the machine learning architecture described herein, which is referred to as RoomNet.

FIG. 1C is another example room layout estimation with a RoomNet.

FIG. 2 shows example definitions of room layout types. The type can be indexed from 0 to 10. The number on each keypoint defines a specific order of points saved in the ground truth. For a given room type, the ordering of the keypoints can specify their connectivity.

FIG. 3 depicts another example architecture of a RoomNet.

FIG. 4A shows an example illustration of an unroller version of a recurrent neural network (RNN) with three iterations.

FIG. 4B shows an example RoomNet with a memory augmented recurrent encoder-decoder (MRED) architecture that mimics the behavior of a RNN but which is designed for a static input.

FIG. 5A-5D show images illustrating example room layout keypoint estimation from single images (middle row) without refinement (top row) and with refinement (bottom row). Keypoint heat maps from multiple channels are shown in a single two dimensional (2D) image for visualization purposes.

FIGS. 6A-6B depicts examples memory augmented recurrent encoder-decoder architectures without deep supervision through time (FIG. 6A) and with deep supervision through time (FIG. 6B).

FIGS. 7A-7G include images showing example RoomNet predictions and the corresponding ground truth on the Large-scale Scene Understanding Challenge (LSUN) dataset. A RoomNet accessed an RGB image as its input (first column in each figure) and produced an example room layout keypoint heat map (second column in each figure). The final keypoints were obtained by extracting the keypoint location having the maximum response from the heat map. The third and fourth columns in each figure show example boxy room layout representations generated by connecting the obtained keypoints in a specific order as described with reference to FIG. 2. The fifth and sixth columns in each figure show example ground truth.

FIGS. 8A-8D show examples where the room layout predictions from an embodiment of RoomNet are less good matches to the (human-annotated) ground truth layouts. The first column in each figure shows an example input image. The second column in each figure shows an example predicted keypoint heat map. The third and fourth columns in each figure show example boxy representations obtained. The fifth and sixth columns show example ground truth.

FIGS. 9A-9F depict example encoder-decoder architectures: (FIG. 9A) a vanilla encoder-decoder; (FIG. 9B) a stacked encoder-decoder; (FIG. 9C) a stacked encoder-decoder with skip-connections; (FIG. 9D) an encoder-decoder with feedback; (FIG. 9E) a memory augmented recurrent encoder-decoder; and (FIG. 9F) a memory augmented recurrent encoder-decoder with feedback.

FIG. 10 is a flow diagram of an example process of training a RoomNet.

FIG. 11 is a flow diagram of an example process of using a RoomNet for room layout estimation.

FIG. 12 schematically illustrates an example of a wearable display system, which can implement an embodiment of RoomNet.

Throughout the drawings, reference numbers may be re-used to indicate correspondence between referenced elements. The drawings are provided to illustrate example embodiments described herein and are not intended to limit the scope of the disclosure.

DETAILED DESCRIPTION Overview

Models representing data relationships and patterns, such as functions, algorithms, systems, and the like, may accept input, and produce output that corresponds to the input in some way. For example, a model may be implemented as a machine learning method such as a convolutional neural network (CNN) or a deep neural network (DNN). Deep learning is part of a broader family of machine learning methods based on the idea of learning data representations as opposed to task specific algorithms and shows a great deal of promise in solving audio-visual computational problems useful for augmented reality, mixed reality, virtual reality, and machines intelligence. In machine learning, a convolutional neural network (CNN, or ConvNet) can include a class of deep, feed-forward artificial neural networks, and CNNs have successfully been applied to analyzing visual imagery. Machine learning methods include a family of methods that can enable robust and accurate solutions to a wide variety of problems, including eye image segmentation and eye tracking.

Disclosed herein are examples of a neural network for room layout estimation called RoomNet. RoomNet can analyze an image of at least a portion of a room to determine the room layout. The room layout can include a representation of locations of a floor, a wall, or a ceiling in the room. The image can, for example, comprise a monocular image or a grayscale or color (e.g., Red-Green-Blue (RGB)) image. The image may be a frame or frames from a video. Other techniques divide room layout estimation into two sub-tasks: semantic segmentation of floor, walls, and ceiling to produce layout hypotheses, followed by iterative optimization step to rank these hypotheses.

In contrast to these approaches, RoomNet can formulates the room layout problem as estimating an ordered set of room layout keypoints. The room layout and the corresponding segmentation can be completely specified given the locations of these ordered keypoints. The RoomNet can be an end-to-end trainable encoder-decoder network. A RoomNet machine learning architecture may have better performance (e.g., in terms of the amount of computation, accuracy, etc.). In some embodiments, a RoomNet can have an architecture that includes recurrent computations and memory units to refine the keypoint locations under similar, or identical, parametric capacity.

Stereoscopic images can provide depth information on a room layout. Room layout estimation from a monocular image (which does not include depth information) is challenging. Room layout estimation from monocular images, which aims to delineate a two-dimensional representation (2D) representation (e.g., boxy representation) of an indoor scene, has applications for a wide variety of computer vision tasks, such as indoor navigation, scene reconstruction or rendering, or augmented reality. FIG. 1A illustrates a conventional room layout technique that takes an image 104, extracts image features 108, such as local color, texture, and edge cues in a bottom-up manner, followed by vanishing point detection 112. Conventional methods may include a separate post-processing stage used to clean up feature outliers and generate, or rank, a large set of room layout hypotheses 116 with structured support vector machines (SVMs) or conditional random fields (CRFs). In principle, the 3D reconstruction of the room layout can be obtained (e.g., up to scale) with knowledge of the 2D layout 120 a and the vanishing points determined using these methods. However, in practice, these conventional methods are complicated and the accuracy of the final layout prediction often largely depends on the quality of the extracted low-level image features, which in itself is susceptible to local noise, scene clutter and occlusion. Advantageously, embodiments of a RoomNet of the disclosure may not be susceptible to local noise, scene clutter and occlusion. Further, room layout estimation provided by RoomNet may advantageously have better performance (e.g., in terms of the amount of computation, such as 200× or 600×) than other methods.

In some embodiments, a RoomNet may have better performance than other room layout estimation methods based on convolutional neural networks (CNNs), such as deep neural networks, semantic segmentation, a fully convolutional network (FCN) model that produces informative edge maps that replace hand engineered low-level image feature extraction. The predicted edge maps generated by such FCN can then be used to sample vanishing lines for layout hypotheses generation and ranking. For example, the FCN can be used to learn semantic surface labels, such as left wall, front wall, right wall, ceiling, and ground. Then connected components and hole filling techniques can be used to refine the raw per pixel prediction of the FCN, followed by the classic vanishing point/line sampling methods to produce room layouts. In contrast to such methods that generate a new set of low-level features and may require 30 seconds or more to process each frame, a RoomNet can be an end-to-end trainable CNN that is more computationally efficient.

In some embodiments, predictions of a RoomNet need not be post-processed by a hypotheses testing stage, which can be expensive, to produce the final layout. A RoomNet may perform room layout estimation using a top-down approach and can be directly trained to infer both the room layout keypoints (e.g., corners) and room type. Once the room type is inferred or determined and the corresponding set of ordered keypoints are localized or determined, the keypoints can be connected in a specific order, based on the room type determined, to obtain the 2D spatial room layout.

A RoomNet architecture may be direct and simple as illustrated in FIGS. 1B and 1C. As will be further explained below, the RoomNet 124 can take an input image 104 (e.g., of size 320 pixels×320 pixels), process the image through a convolutional encoder-decoder architecture, extract a set of room layout keypoints 128 k 1-128 k 6 from a keypoint heat map 128 corresponding to a particular room layout, and then (optionally) connect the obtained keypoints in a specific order to provide a room layout 120 b. The room layout 120 b can include locations or orientations of vertical or horizontal surfaces in the room such as, e.g., a floor 132, a ceiling 134, and walls 136.

Optionally, the room layout can be regressed as described below. The room layout 120 b can be used, for example, in a world map for augmented reality or indoor autonomous navigation or for scene reconstruction or rendering. Optionally, the room layout can be output as a drawing, architectural map, etc. The semantic segmentation of the layout surfaces can be simply obtainable as a consequence of this connectivity and represented as a semantically segmented room layout image 136. Accordingly, a RoomNet performs the task of room layout estimation by keypoint localization. In some embodiments, a RoomNet can be an encoder-decoder network based on a CNN. A RoomNet can be parametrically efficient and effective in joint keypoint regression and room layout type classification.

Example Keypoint-Based Room Layout Representation

Embodiments of a RoomNet can be effective in room layout estimation. A RoomNet can be based on target output representation that is end-to-end trainable and can be inferred efficiently. A RoomNet can complement, or supplement, methods based on assigning geometric context or semantic classes (e.g., floor, walls, or ceiling, etc.) to each pixel in an image, and then obtaining room layout keypoints and boundaries based on the pixel-wise labels. Deriving layout keypoints and boundaries from the raw pixel output may be non-trivial and less efficient than embodiments of a RoomNet. In contrast, a RoomNet can be based on a model that directly outputs a set of ordered room layout keypoint locations, such that both keypoint-based and pixel-based room layout representations may be obtained efficiently with high accuracy. A RoomNet can reduce or eliminate the ambiguity in the pixel-based representation used by other methods. Embodiments of RoomNet thus are able to distinguish between different surface identities (e.g., front walls, side walls, floors, ceilings). For instance, a RoomNet may correctly distinguish between a front wall class and a right wall class, and thereby output regular, not mixed, labels within the same surface. Accordingly, a RoomNet may have better overall room layout estimation accuracy and performance.

In some implementations, a RoomNet may be trained using a keypoint-based room layout representation illustrated in FIG. 2. FIG. 2 shows a list of example room types 0-10 204 rt 0-204 rt 10 with their respective keypoint definition labeled as 1, 2, 3, 4, 5, 6, 7, and/or 8. The number on each keypoint defines a specific order of points saved in ground truth. These 11 room layout types can cover most of the possible situations under typical camera poses and common room layout representations under the Manhattan world assumption, in which objects, edges, corners in images are built on a Cartesian grid, leading to regularities in image gradient statistics. In various embodiments, the room type can be represented by a plurality of polygonal regions, with each region corresponding to, e.g., a floor, a ceiling, right wall, a middle wall, a left wall, etc. The room types can be organized by a set of corner keypoints, for example, corners that correspond to intersections of the polygonal regions. For example, in room type 204 rt 5, a left wall is bounded by keypoints 1, 2, 5, and 4; a right wall keypoint is bounded by keypoints 1, 3, 6, and 4; a floor is bounded by keypoints 5, 4, 6; and a ceiling is bounded by keypoints 2, 1, 3. The room type can be segmented semantically to identify the floor, wall, and ceiling.

Once the trained RoomNet predicts correct keypoint locations with an associated room type, these points can then be connected in a specific order to produce a boxy room layout representation. For example, the room type 7 204 rt 7 includes four ordered keypoint locations 208 k 1-208 k 4, such that a boxy room layout representation can be constructed by connecting keypoint 1 208 k 1 with keypoint 2 208 k 2 and keypoint 3 208 k 3 with keypoint 4 208 k 4. The 11 room layouts include one room layout type 204 rt 0 with eight keypoints, three room layout types 204 rt 1, 204 rt 2, and 204 rt 5 with six keypoints, four room layout types 204 rt 3, 204 rt 4, 204 rt 6, and 204 rt 7 with four keypoints, and three room layout types 204 rt 8, 204 rt 9, and 204 rt 10 with two keypoints. Room layouts with the same number of keypoints can have the same keypoint connectivity (such as room layout type 3 and 4, 204 rt 3 and 204 rt 4,) or different keypoint connectivity (such as room layout type 1 and 2, 204 rt 3 and 204 rt 4). Although 11 room layout types are used in this example, a different number of room layout types can be used in other implementations (e.g., 5, 10, 15, 20, or more) or room layout types having a different arrangement than shown in FIG. 2.

Example Architecture of RoomNet

A neural network for room layout estimation of the disclosure can include a convolutional neural network (CNN) that to delineate room layout structure using two dimensional (2D) keypoints. The input to the RoomNet can be a monocular image, for example, a single Red-Green-Blue (RGB) image or RGB frame from a video. The output of the RoomNet can include a set of 2D keypoints associated with a specific order with an associated room type.

Keypoint estimation. In some embodiments, a RoomNet can include a base network architecture for keypoint estimation and semantic segmentation of surfaces of a room, such as roof (or ceiling), left wall, right wall, back wall, floor, etc. FIG. 3 depicts an example architecture of a RoomNet 300. In this example architecture, a decoder upsamples its input using the transferred pooling indices from its encoder to produce sparse feature maps followed by a several convolutional layers with trainable filter banks to densify the feature responses. The final decoder output keypoint heat maps are fed to a regressor with Euclidean losses. A side head with three fully-connected layers is attached to the bottleneck layer and used to train and predict the room type class label, which is then used to select the associated set of keypoint heat maps. The full model of a RoomNet with recurrent encoder-decoder (center dashed block) further performs keypoint refinement as described with reference to FIGS. 4B and 5.

With continued reference to FIG. 3, the RoomNet 300 can include an encoder sub-network 304 a and a decoder sub-network 304 b. The encoder sub-network 304 a can map an input image 308 to lower resolution feature maps 312 a-312 e. The decoder sub-network 304 b can upsample the low resolution encoded feature maps 312 e to higher resolution maps 316 a-316 b and heat maps 320 r 0-320 r 10 (e.g., with the same or lower resolution compared to the input image 308) for pixel-wise classification. Dimensionalities of the input image 308, feature maps 312 a-312 e, 316 a-316 b, and heat maps 320 r 0-320 r 10 are labeled in the RoomNet example 300 shown in FIG. 3. The encoder sub-network 304 a can include a plurality of convolutional layers and pooling layers 324 a-324 e. The decoder sub-network 304 b can include a plurality of convolutional layers and upsampling layers 328 a-328 c. In some embodiments, the decoder sub-network 304 b can use pooling indices computed in the maxpooling step or layer of the corresponding encoder sub-network 304 b to perform non-linear upsampling. For example, the weights of the maxpooling layer used to generate the feature maps 312 e can be used to upsample the feature maps 312 e. As another example, the weights of the maxpooling layer used to generate the feature maps 312 c can be used to upsample the feature maps 316 a. Pooling indices can minimize, or eliminate, the need for learning to upsample. The upsampled maps can be sparse and can be convolved with trainable filters to produce dense feature maps 316 a, 316 b. This encoder-decoder architecture can provide good performance with competitive inference time and efficient memory usage as compared to other methods for room layout estimation. The number of heat maps 320 r 0-320 r 10 can be the number of defined room types, such as 5, 10, 11, 15, or more. FIG. 3 shows the number of keypoints associated with each room type. For example, room type 0 320 r 0 is associated with eight keypoints. Each of the eight keypoints can be, for example, identified as the highest peak in each of the eight heat maps 320 r 0. Accordingly, the number of heat maps 320 r 0-320 r 10 output by the RoomNet 300 can be the total number of keypoints of the different room types. In the example illustrated in FIG. 3, the number of heat maps 320 r 0-320 r 10 is 48.

The base architecture of the RoomNet 300 can take an image 308 of an indoor scene and directly output a set of 2D room layout keypoints to recover the room layout structure. Each keypoint ground truth can be represented by a 2D Gaussian heat map centered at the true keypoint location as one of the channels in the output layer. In some embodiments, the keypoint heat maps 320 r 0-320 r 10 in a single 2D image can be color coded for visualization. The encoder-decoder architecture of the RoomNet 300 can process the information flow through bottleneck layer (e.g., the convolutional and maxpooling layer 324 e), enforcing the bottleneck layer to implicitly model the relationship among the keypoints that encode the 2D structure of the room layout.

In some embodiments, the decoder sub-network 304 b of the RoomNet 300 can upsample the feature maps 312 e from the bottleneck layer 324 e with spatial dimension 10×10 to 40×40 instead of the full resolution 320 pixels×320 pixels as shown in FIG. 3. Such reduction in the dimensionality of the output heat maps 320 r 0-320 r 10 to 40 pixels×40 pixels, compared to the dimensionality of the input image 308, can be empirically determined using the proposed 2D keypoint-based representation to already model the room layout effectively. In some embodiments, the width and height of heat maps 320 r 0-320 r 10 can be the same as those of the input image 308, such as 320 pixels×320 pixels. Embodiments of the RoomNet 300 with different output dimensions may have similar performance. Using this trimmed decoder sub-network 304 b can advantageously reduce (e.g., significantly reduce) the memory usage or time cost during both training and testing due to the high computation cost of convolution at higher resolutions.

Extending to multiple room types. The framework or architecture of the RoomNet 300 is not limited to one particular room type. Embodiments of the RoomNet can be generalized for multiple room types without training one network per class. Such embodiments of the RoomNet 300 can be efficient and fast from the ground up. The RoomNet embodiment 300 illustrated in FIG. 3 can predict room layout keypoints for an associated room type with respect to the input image in one forward pass. The number of channels in the output layer 328 c can match the total number of keypoints for all defined room types (e.g., a total 48 keypoints for the 11 room types illustrated in FIG. 2). The RoomNet 300 can also include a side head or side sub-network 304 c with connected layers 332 a-332 c (e.g., fully connected layers) to the bottleneck layer 324 e (e.g., a layer usually used for image classification) to predict the room type prediction as shown in FIG. 3. The side sub-network can comprise a classifier network to classify a room type in the room image.

A training example or room image can be denoted as (I, y, t), where y is a list of the ground truth coordinates of the k keypoints with the room type t for the input image I. At the training stage, a loss function L can include a first loss for the predicted keypoints and a second loss for the predicted room type. The first loss can be a Euclidean loss, which can be used as the cost function for layout keypoint heat map regression. During training, the second loss can be a cross-entropy loss (e.g., logarithmic), which can be used for the room type prediction. Given the keypoint heat map regressor φ (e.g., output from the decoder sub-network 304 b), and the room type classifier ψ (e.g., output from the fully-connected side head layer 304 c), the loss function L shown in Equation [1] can be optimized (e.g., reduced or minimized).

L=Σ _(k)

_(k,t) ^(keypoint) ∥G _(k)(y)−φ_(k)(I)∥²−λΣ_(c)

_(c,t) ^(room) log(ψ_(c)(I)),   Equation [1]

where

_(k,t) ^(keypoint) denotes whether keypoint k appears in ground truth room type t,

_(c,t) ^(room) denotes whether room type index c equals to the ground truth room type t, the function G is a Gaussian centered at y, and the weight term is λ. For example, the weight term λ (e.g., 5) can be set by cross validation. The first term in the loss function compares the predicted heat maps 320 r 0-320 r 10 to ground-truth heat maps synthesized for each keypoint separately. The ground truth for each keypoint heat map can be a 2D Gaussian centered on the true keypoint location with standard deviation of a number of pixels (e.g., 5 pixels). The second term in the loss function can encourage the side head 304 c fully-connected layers 332 a-332 c to produce a high confidence value with respect to the correct room type class label.

One forward pass of the RoomNet 300 can produce 2D room layout keypoints 320 r 0-320 r 10 for all defined room types (e.g., 11 in FIG. 2). The 2D room layout keypoints can be in the form of heat maps, where the final keypoints can be extracted as the maxima in these heat maps. In some embodiments, the loss function (e.g., the loss function L shown in Equation [1]) only penalizes Euclidean regression error if the keypoint k is present for the ground truth room type t in the current input image I, effectively using the predicted room type indices to select the corresponding set of keypoint heat maps to update the regressor. The same strategy can apply after the RoomNet 300 is trained (e.g., at the test stage) such that the predicted room type (e.g., by the side network 304 c) is used to select the predicted keypoint heat map in the final output.

RoomNet extension for keypoint refinement. Recurrent neural networks (RNNs) and their variants Long Short-Term Memory (LSTM) can be effective models when dealing with sequential data. Embodiments of a RoomNet 300 can incorporate recurrent structures, even though the input image 308 is static. For example, a RoomNet 300 can include recurrent convolutional layers and convolutional LSTM (convLSTM) layers. In some embodiments, recurrent features of a RoomNet 300 can be similar to models such as a fully convolutional network (FCN) with conditional random fields as recurrent neural network (CRF-RNN), iterative error feedback networks, recurrent CNNs, stacked encoder-decoder, and recurrent encoder-decoder networks. Incorporating a time series concept when modeling a static input can significantly improve the ability of the RoomNet 300 to integrate contextual information and to reduce prediction error in some cases.

A base RoomNet architecture can be extended by making the central encoder-decoder component 336 (see, e.g., the center dashed line block in FIG. 3) recurrent. For example, a RoomNet 300 can include a memory augmented recurrent encoder-decoder (MRED) structure 404 b (see FIG. 4B) to mimic the behavior of a typical recurrent neural network 404 a (see the example shown in FIG. 4A) in order to refine the predicted keypoint heat maps over by iterating over an artificial time—the artificial time steps (e.g., the iterations) are created by the recurrent structure.

Each layer 312 c-312 e, 316 a-316 b in this MRED structure 404 b can share the same weight matrices through different time steps (e.g., iterations) that convolve (denoted as * symbol) with the incoming feature maps from the previous prediction h_(l)(t−1) at time step t−1 in the same layer l, and the current input h_(l−)1(t) at time step t in the previous layer l−1, generating output at time step t as shown in Equation [2].

$\begin{matrix} {{h_{l}(t)} = \left\{ {\begin{matrix} {{{\sigma \left( {{w_{l}^{current}*{h_{l - 1}(t)}} + b_{l}} \right)},}\ } & {t = 0} \\ {{{\sigma \left( {{w_{l}^{current}*{h_{l - 1}(t)}} + {w_{l}^{{pre\nu}i{ous}}*{h_{l}\left( {t - 1} \right)}} + b_{l}} \right)}\ ,}\ } & {t > 0} \end{matrix},} \right.} & {{{Equation}\mspace{14mu}\lbrack 2\rbrack},} \end{matrix}$

where w_(l) ^(current) and w_(l) ^(previous) are the input and feed-forward weights for layer l, b_(l) is the bias for layer l, and σ is an activation function, e.g., a rectified linear unit (ReLU) activation function.

FIG. 4B demonstrates an example overall process of the information flow during forward propagations and backward propagations through depth and time within the recurrent encoder-decoder structure. The memory augmented recurrent encoder-decoder (MRED) architecture 404 b includes hidden units 408 a, 408 b to store previous activations that help the inference at the current time step. Non-limiting example advantages of using the proposed MRED 404 b architecture include (1) exploiting the contextual and structural knowledge among keypoints iteratively through hidden/memory units (e.g., that have not been explored in recurrent convolutional encoder-decoder structure) or (2) weight sharing of the convolutional layers in the recurrent encoder-decoder, resulting in a much deeper network with a fixed number of parameters.

After refinement, the heat maps of keypoints are much cleaner as shown in the bottom rows of FIG. 5A-5D. FIGS. 5A-5D show images illustrating example room layout keypoint estimation from single images (middle row, images 504 a-504 d) without refinement (top row, heat maps 508 a-508 d) and with refinement (bottom row, heat maps 512 a-512 d). Keypoint heat maps from multiple channels are shown in a single two dimensional (2D) image for visualization purposes. The keypoint refinement step produces more concentrated and cleaner heat maps and removes false positives, if any. Improvements were made by embodiments of the RoomNet 300 with an MRED architecture 404 b (see FIGS. 5C-5D).

Deep supervision through time. When applying stacked, iterative, or recurrent convolutional structures, each layer in a network can receive gradients across more layers or/and time steps, resulting in models that are much harder to train. For instance, the iterative error feedback network can require multi-stage training and the stacked encoder-decoder structure can uses intermediate supervision at the end of each encoder-decoder even when batch normalization is used. Training a RoomNet 300 can include injecting supervision at the end of each time step. For example, the same loss function L 604, such as the loss function shown in Equation [1], can be applied to all the time steps. The three loss functions L₁ 604 a, L₂ 604 b, and L₃ 604 c that are injected at the end of each time step in FIG. 6B can be the identical or different. FIGS. 6A-6B depict examples of memory augmented recurrent encoder-decoder architectures without deep supervision through time (FIG. 6A) and with deep supervision through time (FIG. 6B). Deep supervision can improve performance of a RoomNet 300 through time.

Example Training

Datasets. Embodiments of a RoomNet 300 were tested on two challenging benchmark datasets: the Hedau dataset and the Large-scale Scene Understanding Challenge (LSUN) room layout dataset. The Hedau dataset contains 209 training, 53 validation, and 105 test images that are collected from the web and from LabelMe. The LSUN dataset consists of 4000 training, 394 validation, and 1000 test images that are sampled from SUN database. All input images were rescaled to 320×320 pixels and used to train the RoomNet 300 from scratch on the LSUN training set only. All experimental results were computed using the LSUN room layout challenge toolkit on the original image scales.

Implementation details. The input to the RoomNet 300 was an RGB image of resolution 320×320 pixels and the output was the room layout keypoint heat maps of resolution 40×40 with an associated room type class label. In other implementations, the image resolution or the heat map resolution can be different. Backpropagation through time (BPTT) algorithm was applied to train the models with batch size 20 stochastic gradient descent (SGD), 0.5 dropout rate, 0.9 momentum, and 0.0005 weight decay. Initial learning rate was 0.00001 and decreased by a factor of 5 twice at epoch 150 and 200, respectively. All variants used the same scheme with 225 total epochs. The encoder and decoder weights were initialized. Batch normalization and rectified linear unit (ReLU) activation function were also used after each convolutional layer to improve the training process. Horizontal flipping of input images was used during training as data augmentation. In some embodiments, a RoomNet 300 can be implemented in the open source deep learning framework Caffe.

A ground truth keypoint heat map may have zero value (background) for most of its area and only a small portion of it corresponds to the Gaussian distribution (foreground associated with actual keypoint location). The output of the network therefore may tend to converge to zero due to the imbalance between foreground and background distributions. In some embodiments, the gradients were weighted based on the ratio between foreground and background area for each keypoint heat map. Gradients of background pixels were degraded by multiplying them with a factor of 0.2, which made training significantly more stable. In some cases, pixels in the background comprise the pixels that are farther from a keypoint than a threshold distance, for example, the standard deviation of the Gaussian distribution used to generate the ground truth heat map, e.g., greater than 5 pixels.

Training from scratch took about 40 hours on 4 NVIDIA Titan X GPUs for one embodiment of RoomNet. One forward inference of the full model (RoomNet recurrent 3-iteration) took 83 ms on a single GPU. For generating final test predictions, both the original input and a flipped version of the image were ran through the network and the heat maps were averaged together (accounting for a 0.12% average improvement on keypoint error and a 0.15% average improvement on pixel error). The keypoint location was chosen to be the max activating location of the corresponding heat map.

Example Performance

In some embodiments, room layout estimation evaluation metrics can include: pixel errors and keypoint errors. A pixel error can be a pixel-wise error between the predicted surface labels and ground truth labels. A keypoint error can be an average Euclidean distance between the predicted keypoint and annotated keypoint locations, normalized by the image diagonal length.

Accuracy. The performance of a RoomNet 300 on both datasets are listed in Table 1 and 2. The previous best method was the two-step framework (per pixel CNN-based segmentation with a separate hypotheses ranking approach). The RoomNet 300 of the disclosure can significantly improve upon and outperform the previous results on both keypoint error and pixel error, achieving state-of-the-art performance. The side head room type classifier obtained 81.5% accuracy on LSUN dataset.

TABLE 1 Performance of a RoomNet architecture on the Hedau dataset. Method Pixel Error (%) Hedau et al. (2009) 21.20 Del Pero et al. (2012) 16.30 Gupta et al. (2010) 16.20 Zhao et al. (2013) 14.50 Ramalingam et al. (2013) 13.34 Mallya et al. (2015) 12.83 Schwing et al. (2012) 12.8 Del Pero et al. (2013) 12.7 Dasgupta et al. (2016) 9.73 RoomNet recurrent 3-iteration 8.36

TABLE 2 Performance of a RoomNet architecture on LSUN dataset. Keypoint Pixel Method Error (%) Error (%) Hedau et al. (2009) 15.48 24.23 Mallya et al. (2015) 11.02 16.71 Dasgupta et al. (2016) 8.20 10.63 RoomNet recurrent 6.30 9.86 3-iteration

TABLE 3 Runtime evaluation of a RoomNet on an input size of 320 pixels × 320 pixels. The RoomNet full model (with 3- iterations in time) achieved 200 times speedup and the basic RoomNet model (without any iteration in time) achieved 600 times speedup as compared to other methods. Method FPS Del Pero et al. (2013) 0.001 Dasgupta et al. (2016) 0.03 RoomNet recurrent 3-iter 5.96 RoomNet recurrent 2-iter 8.89 RoomNet basic (no iterations) 19.26

TABLE 4 The impact of keypoint refinement step using the memory augmented recurrent encoder- decoder architecture on the LSUN dataset. Keypoint Pixel Method Error (%) Error (%) RoomNet basic 6.95 10.46 RoomNet recurrent 6.65 9.97 2-iterations RoomNet recurrent 6.30 9.86 3-iterations

TABLE 5 The impact of deep supervision through time on LSUN dataset for RoomNets with 2 and 3 recurrent iterations. Keypoint Pixel Model Error (%) Error (%) RoomNet recurrent 2-iteration w/o deep supervision through time 6.93 10.44 w/ deep supervision through time 6.65 9.97 RoomNet recurrent 3-iteration w/o deep supervision through time 6.95 10.47 w/ deep supervision through time 6.30 9.86

Runtime and complexity. Efficiency evaluation on the input image size of 320×320 is shown in Table 3. The full model (RoomNet recurrent 3 iteration) achieved 200× speedup compares another method of room layout estimation, and the base RoomNet without recurrent structure (RoomNet basic) achieved 600× speedup. The timing was for two forward passes as described herein. Using either one of the proposed RoomNet 300 can provide significant inference time reduction and an improved accuracy as shown in Table 4.

Example RoomNet Analysis

Recurrent vs. direct prediction. The effect of each component in the RoomNet architecture was investigated with the LSUN dataset. Table 4 shows the effectiveness of extending the RoomNet basic architecture to a memory augmented recurrent encoder-decoder networks. It was observed that more iterations led to lower error rates on both keypoint error and pixel error: the RoomNet 300 with recurrent structure that iteratively regressed to correct keypoint locations achieved 6.3% keypoint error and 9.86 pixel error as compared to the RoomNet 300 without recurrent structure which achieved 6.95% keypoint error and 10.46 pixel error. No further significant performance improvement was observed after three iterations. Without being limited by the theory, the improvement may come from the same parametric capacity within the networks since the weights of convolutional layers are shared across iterations.

Effect of deep supervision through time. When applying a recurrent structure with an encoder-decoder architecture, each layer in the network receives gradients not only across depth but also through time steps between the input and the final objective function during training. The effect of adding auxiliary loss functions at different time steps was determined. Table 5 demonstrates the impact of deep supervision through time using RoomNet 300 with two or three recurrent iterations. Immediate reduction in both keypoint error and pixel error by adding auxiliary losses for both cases. In some embodiments, the learning problem with deep supervision can be easier through different time steps. The RoomNet 300 with three iterations in time performed worse than RoomNet 300 with two iterations when deep supervision through time was not applied. This was rectified when deep supervision through time was applied. In some embodiments, with more iterations in the recurrent structure, deep supervision through time can be applied to successfully train the architecture.

Qualitative results. Qualitative results of the RoomNet 300 are shown in FIGS. 7A-7G. FIGS. 7A-7G are images showing example RoomNet predictions and the corresponding ground truth on the Large-scale Scene Understanding Challenge (LSUN) dataset. A RoomNet took an RGB image as its input 704 a-704 g (drawn in the first column in each figure) and produced an example room layout keypoint heat map 708 a-708 g (second column in each figure). The final keypoints were obtained by extracting the location with maximum response from the heat map. The third and fourth columns in each figure show example boxy room layout representations 712 a-712 fg 716 a-716 g by connecting the obtained keypoints in a specific order as in FIG. 2. The different surfaces in the third column are shown in different cross-hatch patterns, which can result from a segmentation of the layout to identify a ceiling, a floor, walls, etc. The RoomNet room layout output 712 a-712 g shows the floor, ceiling, and walls in different cross-hatches. In representations 716 a-716 g, the room layout is superimposed on the respective input image 704 a-704 g. The fifth and sixth columns in each figure show example ground truths 720 a-720 g, 724 a-724 g for the actual room layouts. The correspondences between the room layouts 712 a-712 g and 716 a-716 g (determined by RoomNet) and the actual ground truth layouts 720 a-720 g and 724 a-724 g is striking. These example results demonstrate that RoomNet is robust to keypoint occlusion by objects (e.g., tables, chairs, beds, etc.). When the image was clean and the room layout boundaries/corners were not occluded, the RoomNet 300 can recover the boxy room layout representation with high accuracy. The RoomNet framework was also robust to keypoint occlusion by objects (e.g., tables, chairs, beds, etc.), demonstrated in, e.g., FIGS. 7B, 7C, 7D, 7F.

FIGS. 8A-8D are example images showing examples where the room layout predictions from an embodiment of RoomNet are less good matches to the ground truth layouts. The differences between the RoomNet predictions and the ground truth can be further reduced or eliminated as described herein. The first column in each figure shows an example input image 804 a-804 d. The second column in each figure shows an example predicted keypoint heat map 808 a-808 d. The third and fourth columns in each figure show example boxy representations obtained 812 a-812 d, 816 a-816 d. The different surfaces in the third column are shown in different cross-hatch patterns, which can result from a segmentation of the layout to identify a ceiling, a floor, walls, etc. The fifth and sixth columns show example ground truths 820 a-820 d, 824 a-824 d. Further improvements of the RoomNet 300 may be possible when room layout boundaries are barely visible (e.g., FIGS. 8A and 8C), or when there is more than one plausible room layout for a given image of a scene (e.g., FIGS. 8B and 8D).

Example Alternative Encoder-Decoder

The effect of each component in the proposed architecture with the LSUN dataset was empirically determined. An evaluation of six alternative encoder-decoder architectures shown in FIGS. 9A-9F for the room layout estimation task investigated included: (a) a vanilla encoder/decoder 900 a (RoomNet basic), shown in FIG. 9A; (b) a stacked encoder-decoder 900 b, shown in FIG. 9B, (c) a stacked encoder-decoder with skip-connections 900 c, shown in FIG. 9C; (d) an encoder-decoder with feedback 900 d, shown in FIG. 9D; (e) memory augmented recurrent encoder-decoder (RoomNet full) 900 e, shown in FIG. 9E; and (f) a memory augmented recurrent encoder-decoder with feedback 900 f, shown in FIG. 9F. Some embodiments of the RoomNet 300 may have advantages over other embodiments of the RoomNet 300 for certain tasks; for example, some embodiments of RoomNet can reduce or eliminate the differences shown in FIGS. 8A-8D. Table 6 shows the performance of different variants on LSUN dataset.

TABLE 6 Evaluation of encoder-decoder (enc-dec) variants on LSUN dataset. Keypoint Pixel Model Error (%) Error (%) Vanilla enc-dec (RoomNet basic) 6.95 10.46 Stacked enc-dec 6.82 10.31 Stacked enc-dec with skip connect. 7.05 10.48 Enc-dec w/ feedback 6.84 10.10 Recurrent enc-dec (RoomNet full) 6.30 9.86 Recurrent enc-dec w/ feedback 6.37 9.88 Note that recurrent encoder-decoders use three iteration time steps.

The comparison of the (a) and (b) configurations 900 a, 900 b indicates that stacking encoder-decoder networks can further improve the performance, as the network is enforced to learn the spatial structure of the room layout keypoints implicitly by placing constraints on multiple bottleneck layers.

However, adding skip connections as in the (c) configuration 900 c did not improve the performance for this task under the conditions tested. This could be because the size of the training set (thousands) was not as large as other datasets (millions) that had been evaluated on, therefore skipping layers was not necessary for the specific dataset.

Adding a feedback loop, implemented as a concatenation of input and previous prediction as a new input for the same encoder-decoder network as in the (d) configuration 900 d improved the performance. At each iteration, the network had access to the thus-far sub-optimal prediction along with the original input to help inference at the current time step.

Making an encoder-decoder recurrent with memory units in the (e) configuration 900 e to behave as a RNN obtains the lowest keypoint error and pixel error (the full RoomNet model). The lateral connections in the recurrent encoder-decoder allowed the network to carry information forward and help prediction at future time steps. Adding a feedback loop to the memory augmented recurrent encoder-decoder in the (f) configuration 900 f did not improve the results. It was possible that using the memory augmented structure in the configuration (e) 900 e can already store previous hidden state information well without feedback. Weight matrices of the encoder-decoder were not shared in the (b) and (c) configurations 900 b, 900 c but shared in the (d), (e), and (f) configurations 900 d, 900 e, 900 f, resulting in more parametrically efficient architectures.

Feature transferring by pre-training. To decouple the performance gains due to external data, results of fine-tuning the RoomNet from a SUN pre-trained model (on semantic segmentation task) were determined. As shown in Table 7, such a RoomNet achieved 6.09% keypoint error and 9.04% pixel error as compared of other methods with at least 7.95% keypoint error and 9.31% pixel error on the LSUN dataset. Table 7 reflects room layout estimation results with extra data or pre-trained models. In some embodiments, a RoomNet can be trained using an additional Hedau+ training set and fine-tuned from NYUDv2 RGBD (RGB plus Depth) pre-trained models. Table 7 shows the results of fine-tuning from PASCAL and SUN pre-trained RoomNet. The SUN pre-trained RoomNet achieved lowest keypoint error and pixel error on LSUN dataset.

TABLE 7 Evaluation of methods with pre-training techniques on LSUN dataset. Keypoint Pixel Model Error (%) Error (%) Ren et al. 7.95 9.31 Room Net recurrent 3-iterations with PASCAL pre-training 6.43 9.16 With SUN pre-training 6.09 9.04

In some embodiments, a RoomNet 300 can include a gating mechanism to allow incoming signal to alter the state of recurrent units. In some embodiments, a RoomNet 300 can be trained using sequential data and/or predict building room layout maps using sequential data.

Example Process of Training a RoomNet

FIG. 10 is a flow diagram of an example process 1000 of training a RoomNet. The process 1000 can be performed by a hardware processor comprising non-transitory memory configured to store images, the RoomNet architecture and parameters (e.g., NN weights), room types, 2D keypoint locations (e.g., heat maps), room layouts, and so forth.

The process 1000 starts at block 1004, where training room images for many types of rooms and room types are received. Each of the training room images can be associated with a reference room type and a reference keypoints that identify the room layout (e.g., floor, ceiling, wall(s)). In some cases, the training images are annotated by hand to indicate the ground truth (e.g., keypoint location and room type) for the room shown in the image. A training room image can be a monocular image, an Red-Green-Blue (RGB) image, etc. Training images are obtainable from the Hedau dataset or the LSUN dataset.

The process 1000 can include performing a data augmentation strategy (e.g., augmenting the training data with horizontally flipped images) to improve the performance of a trained RoomNet. The number of room types can be different in different implementations, such as 2, 3, 5, 10, 11, 15, 20, or more. A room type can be associated with a plurality of keypoints associated with a keypoint order. The keypoints can be connected in the keypoint order to provide a room layout. The number of keypoints can be different in different implementations, such as 2, 3, 5, 6, 8, 10, 20, 50, or more.

At block 1008, a neural network for room layout estimation (e.g., RoomNet) can be generated. As described herein, an embodiment of RoomNet can comprise: an encoder sub-network, a decoder sub-network connected to the encoder network, and a side head or sub-network connected to the encoder network. The encoder sub-network can comprise a plurality of convolutional layers and a plurality of pooling layers. The decoder sub-network can comprise a plurality of convolutional layers and a plurality of upsampling layers. Weights of a decoder layer of the decoder sub-network can comprise weights of a corresponding encoder layer of the encoder sub-network. Alternatively, or additionally, weights of a decoder layer of the decoder sub-network can be identical to weights of a corresponding encoder layer of the encoder sub-network. In some embodiments, the encoder sub-network and the decoder sub-network comprises a plurality of recurrent layers to form a recurrent encoder-decoder structure (e.g., a memory-augmented recurrent encoder-decoder (MRED) network). A number of recurrent iterations of the recurrent layers can be 2, 3, 5, 10, or more. In some embodiments, weights associated with a first recurrent iteration of the iterations of the recurrent layers are identical to weights associated with a second recurrent iteration of the current layers.

The encoder sub-network and decoder sub-network can have different architectures in different implementations. For example, the encoder sub-network and the decoder sub-network can have a stacked encoder-decoder architecture. As another example, the encoder sub-network and the decoder sub-network can have a stacked encoder-decoder architecture with skip-connections. As yet another example, the encoder sub-network and the decoder sub-network can have a stacked encoder-decoder architecture with feedback. In one example, the encoder sub-network and the decoder sub-network has a memory augmented recurrent encoder-decoder (MRED) architecture. In another example, the encoder sub-network and the decoder sub-network has a memory augmented recurrent encoder-decoder (MRED) architecture with feedback. Feature maps of a RoomNet with a recurrent layer can be determined using Equation [2] in some embodiments.

At block 1012, a plurality of predicted 2D keypoints for each of the room types can be determined using the encoder sub-network and the decoder sub-network of the RoomNet and the training room image. A dimensionality of the training room image can be smaller, the same, or larger than a dimensionality of a predicted heat map. The 2D keypoint locations can in some cases be extracted from heat maps (e.g., as maxima in the heat maps).

At block 1016, a predicted room type can be determined using the encoder sub-network and the side sub-network of the RoomNet and the training room image. For example, side sub-network can comprise a plurality of layers, such as fully-connected layers. In some embodiments, the side sub-network comprises three fully-connected layers. The dimensionality of the output layer of the side sub-network and the number of the plurality of room types can be identical.

At block 1020, the process 1000 can optimize (e.g., reduce or minimize) a loss function based on a first loss representing errors in the predicted keypoints relative to the reference keypoints in the training image and a second loss representing an error in the predicted room type relative to the reference room type in the training image. An example of a loss function L is described with reference to Equation [1]. The first loss can be a Euclidean loss between the predicted keypoints and the reference keypoints. In some implementations, the predicted keypoints are represented by a heat map and a reference heat map for the reference (e.g., ground truth) keypoints can be generated by placing a Gaussian centered on the reference keypoint locations. The first loss can be set up to penalize a predicted keypoint only if the keypoint is present for the reference room type in the input training image. The second loss can be a cross-entropy (e.g., logarithmic) loss based on a room type classifier (e.g., output from the room type side sub-network) that encourages the side sub-network to produce a high confidence value with respect to the correct room type.

In some embodiments, determining the first loss comprises determining a reference heat map using the reference keypoints and determining a difference between the reference heat map and a predicted heat map for the predicted keypoints. The reference heat map can comprise a distribution centered around each reference keypoint location. The distribution can comprise a two-dimensional Gaussian distribution. The Gaussian distribution can have a standard deviation of, for example, 2, 3, 5, 10, or more pixels. The Gaussian distribution can have a standard deviation of a percentage of a dimension of the reference heat map, such as 5%, 10%, 20%, 25%, or more. In some embodiments, determining the reference heat map can comprise degrading values of pixels that are a threshold number of pixels away from the reference keypoint locations, for example, by multiplying the pixel values by a degrading factor less than one, such as 0.1, 0.2, 0.3.

At block 1024, neural network parameters for the RoomNet can be updated based on the optimized loss function. In some embodiments, weights of the RoomNet can be updated by back propagation.

The process 1000 can be iterated for each of the training images in the training image set (e.g., the Hedau or LSUN datasets) to tune the neural network to produce a robust neural network model with reduced or minimized errors (e.g., pixel errors or keypoint errors as described above). An embodiment of RoomNet can be trained using the process 1000 and then applied to real-world images in augmented or mixed reality, indoor navigation, scene reconstruction or rendering, etc.

Example Process of Using a RoomNet for Room Layout Estimation

FIG. 11 is a flow diagram of an example process 1100 of using a RoomNet for estimating a room layout from a room image. The process 1100 can be performed by a hardware processor comprising non-transitory memory configured to store images, the RoomNet architecture and parameters (e.g., NN weights), room layout types, heat maps, room layouts, and so forth. As described below with reference to FIG. 12, the wearable display system 1200 or a robotics system can be configured to implement an embodiment of the process 1100.

The process 1100 starts at block 1104, where a system (e.g., the wearable display system 1200 described with reference to FIG. 12) receives an input image including a possible room scene. The image can include one room scene. The image can comprise a color image (e.g., RGB or RGB-D) and the image may be monocular. The image may be a frame of a video and may be obtained using the outward-facing imaging system 1244 of the wearable display system 1200 described with reference to FIG. 12.

At block 1108, the wearable display system 1200 can access a neural network for room layout estimation (RoomNet), such as the RoomNet trained by the process 1000 illustrated in FIG. 10. The RoomNet can include an encoder sub-network, a decoder sub-network connected to the encoder network, and a side sub-network connected to the encoder network. In some embodiments, the encoder sub-network and the decoder sub-network comprises a plurality of recurrent layers. For example, the encoder-decoder sub-network can comprise a memory-augmented recurrent encoder-decoder (MRED) network. Feature maps of a RoomNet with a recurrent layer can be determined using Equation [2] above. The architectures of the RoomNet can be different in different implementations. For example, the encoder sub-network and the decoder sub-network can comprise a stacked encoder-decoder architecture. As another example, the encoder sub-network and the decoder sub-network can comprise a plurality of skip connections. In one example, the side sub-network comprises a plurality of feedback layers.

At block 1112, the process 1100, using the encoder sub-network and the decoder sub-network of the RoomNet and the room image, can determine a plurality of 2D keypoints corresponding to each of a plurality of room types. The 2D keypoints can be associated with a heat map, and keypoint locations can be extracted from the heat map as maxima that occur in the heat map. The number of room types can be greater than 5 (e.g., 11 in some cases).

At block 1116, the wearable display system 1200 can determine, using the side sub-network of the RoomNet and the room image, a predicted room type from the plurality of room types.

At block 1120, the process 1100 can determine a layout of the room in the room image from the predicted room type and the 2D keypoints. The room layout can comprise ordered keypoints having a keypoint order associated with the predicted room type. The number of the keypoints can be different in different implementations, such as 2, 3, 4, 6, 8, 10, or more. The room layout can comprise a semantic segmentation of layout surfaces such as an identification of a layout surface as a ceiling, a floor, a wall, etc. The semantic segmentation can be derived from the ordered 2D keypoints. Accordingly, the neural network can provide a 3D room layout structure using the 2D keypoints, which can be in a specific order with an associated room type.

At block 1124, the process 1100 can utilize the room layout in an augmented or mixed reality application, for autonomous indoor navigation, for scene reconstruction or rendering, etc.

The wearable display system 1200 (described with reference to FIG. 12 below) can interact with a user of the system based on the predicted layout of the room in the input image. In some embodiments, the wearable display system 1200 can perform indoor navigation based on the predicted layout of a room in the room image, for example, to direct the user to a desired location in the room (e.g., by rendering the room layout and (optionally) a path to the location). In other embodiments, the wearable display system 1200 can reconstruct a scene in the room image based on the predicted layout of the room in the room image.

Example NN Layers

As described above, embodiments of RoomNet can comprise a neural network. A layer of a neural network (NN), such as a deep neural network (DNN) can apply a linear or non-linear transformation to its input to generate its output. A deep neural network layer can be a normalization layer, a convolutional layer, a softsign layer, a rectified linear layer, a concatenation layer, a pooling layer, a recurrent layer, an inception-like layer, or any combination thereof. The normalization layer can normalize the brightness of its input to generate its output with, for example, Euclidean or L2 normalization. The normalization layer can, for example, normalize the brightness of a plurality of images with respect to one another at once to generate a plurality of normalized images as its output. Non-limiting examples of methods for normalizing brightness include local contrast normalization (LCN) or local response normalization (LRN). Local contrast normalization can normalize the contrast of an image non-linearly by normalizing local regions of the image on a per pixel basis to have a mean of zero and a variance of one (or other values of mean and variance). Local response normalization can normalize an image over local input regions to have a mean of zero and a variance of one (or other values of mean and variance). The normalization layer may speed up the training process.

The convolutional layer can apply a set of kernels that convolve its input to generate its output. The softsign layer can apply a softsign function to its input. The softsign function (softsign(x)) can be, for example, (x/(1+|x|)). The softsign layer may neglect impact of per-element outliers. The rectified linear layer can be a rectified linear layer unit (ReLU) or a parameterized rectified linear layer unit (PReLU). The ReLU layer can apply a ReLU function to its input to generate its output. The ReLU function ReLU(x) can be, for example, max(0, x). The PReLU layer can apply a PReLU function to its input to generate its output. The PReLU function PReLU(x) can be, for example, x if x≥0 and ax if x<0, where a is a positive number. The concatenation layer can concatenate its input to generate its output. For example, the concatenation layer can concatenate four 5×5 images to generate one 20×20 image. The pooling layer can apply a pooling function which down samples its input to generate its output. For example, the pooling layer can down sample a 20×20 image into a 10×10 image. Non-limiting examples of the pooling function include maximum pooling, average pooling, or minimum pooling.

At a time point t, the recurrent layer can compute a hidden state s(t), and a recurrent connection can provide the hidden state s(t) at time t to the recurrent layer as an input at a subsequent time point t+1. The recurrent layer can compute its output at time t+1 based on the hidden state s(t) at time t. For example, the recurrent layer can apply the softsign function to the hidden state s(t) at time t to compute its output at time t+1. The hidden state of the recurrent layer at time t+1 has as its input the hidden state s(t) of the recurrent layer at time t. The recurrent layer can compute the hidden state s(t+1) by applying, for example, a ReLU function to its input. The inception-like layer can include one or more of the normalization layer, the convolutional layer, the softsign layer, the rectified linear layer such as the ReLU layer and the PReLU layer, the concatenation layer, the pooling layer, or any combination thereof.

The number of layers in the NN can be different in different implementations. For example, the number of layers in the DNN can be 50, 100, 200, or more. The input type of a deep neural network layer can be different in different implementations. For example, a layer can receive the outputs of a number of layers as its input. The input of a layer can include the outputs of five layers. As another example, the input of a layer can include 1% of the layers of the NN. The output of a layer can be the inputs of a number of layers. For example, the output of a layer can be used as the inputs of five layers. As another example, the output of a layer can be used as the inputs of 1% of the layers of the NN.

The input size or the output size of a layer can be quite large. The input size or the output size of a layer can be n×m, where n denotes the width and m denotes the height of the input or the output. For example, n or m can be 11, 21, 31, or more. The channel sizes of the input or the output of a layer can be different in different implementations. For example, the channel size of the input or the output of a layer can be 4, 16, 32, 64, 128, or more. The kernel size of a layer can be different in different implementations. For example, the kernel size can be n×m, where n denotes the width and m denotes the height of the kernel. For example, n or m can be 5, 7, 9, or more. The stride size of a layer can be different in different implementations. For example, the stride size of a deep neural network layer can be 3, 5, 7 or more.

In some embodiments, a NN can refer to a plurality of NNs that together compute an output of the NN. Different NNs of the plurality of NNs can be trained for different tasks. A processor (e.g., a processor of the local data processing module 1224 descried with reference to FIG. 12) can compute outputs of NNs of the plurality of NNs to determine an output of the NN. For example, an output of a NN of the plurality of NNs can include a likelihood score. The processor can determine the output of the NN including the plurality of NNs based on the likelihood scores of the outputs of different NNs of the plurality of NNs.

Example Wearable Display System

In some embodiments, a user device can be, or can be included, in a wearable display device, which may advantageously provide a more immersive virtual reality (VR), augmented reality (AR), or mixed reality (MR) experience, where digitally reproduced images or portions thereof are presented to a wearer in a manner wherein they seem to be, or may be perceived as, real.

Without being limited by theory, it is believed that the human eye typically can interpret a finite number of depth planes to provide depth perception. Consequently, a highly believable simulation of perceived depth may be achieved by providing, to the eye, different presentations of an image corresponding to each of these limited number of depth planes. For example, displays containing a stack of waveguides may be configured to be worn positioned in front of the eyes of a user, or viewer. The stack of waveguides may be utilized to provide three-dimensional perception to the eye/brain by using a plurality of waveguides to direct light from an image injection device (e.g., discrete displays or output ends of a multiplexed display which pipe image information via one or more optical fibers) to the viewer's eye at particular angles (and amounts of divergence) corresponding to the depth plane associated with a particular waveguide.

In some embodiments, two stacks of waveguides, one for each eye of a viewer, may be utilized to provide different images to each eye. As one example, an augmented reality scene may be such that a wearer of an AR technology sees a real-world park-like setting featuring people, trees, buildings in the background, and a concrete platform. In addition to these items, the wearer of the AR technology may also perceive that he “sees” a robot statue standing upon the real-world platform, and a cartoon-like avatar character flying by which seems to be a personification of a bumble bee, even though the robot statue and the bumble bee do not exist in the real world. The stack(s) of waveguides may be used to generate a light field corresponding to an input image and in some implementations, the wearable display comprises a wearable light field display. Examples of wearable display device and waveguide stacks for providing light field images are described in U.S. Patent Publication No. 2015/0016777, which is hereby incorporated by reference herein in its entirety for all it contains.

FIG. 12 illustrates an example of a wearable display system 1200 that can be used to present a VR, AR, or MR experience to a display system wearer or viewer 1204. The wearable display system 1200 may be programmed to perform any of the applications or embodiments described herein (e.g., estimating a room layout using RoomNet). The display system 1200 includes a display 1208, and various mechanical and electronic modules and systems to support the functioning of that display 1208. The display 1208 may be coupled to a frame 1212, which is wearable by the display system wearer or viewer 1204 and which is configured to position the display 1208 in front of the eyes of the wearer 1204. The display 1208 may be a light field display. In some embodiments, a speaker 1216 is coupled to the frame 1212 and positioned adjacent the ear canal of the user in some embodiments, another speaker, not shown, is positioned adjacent the other ear canal of the user to provide for stereo/shapeable sound control. The display system 1200 can include an outward-facing imaging system 1244 (e.g., one or more cameras) that can obtain images (e.g., still images or video) of the environment around the wearer 1204. Images obtained by the outward-facing imaging system 1244 can be analyzed by embodiments of RoomNet to determine a representation of a room layout in the environment around the wearer 1204.

The display 1208 is operatively coupled 1220, such as by a wired lead or wireless connectivity, to a local data processing module 1224 which may be mounted in a variety of configurations, such as fixedly attached to the frame 1212, fixedly attached to a helmet or hat worn by the user, embedded in headphones, or otherwise removably attached to the user 1204 (e.g., in a backpack-style configuration, in a belt-coupling style configuration).

The local processing and data module 1224 may comprise a hardware processor, as well as non-transitory digital memory, such as non-volatile memory e.g., flash memory, both of which may be utilized to assist in the processing, caching, and storage of data. The data include data (a) captured from sensors (which may be, e.g., operatively coupled to the frame 1212 or otherwise attached to the wearer 1204), such as image capture devices (such as cameras), microphones, inertial measurement units, accelerometers, compasses, GPS units, radio devices, and/or gyros; and/or (b) acquired and/or processed using remote processing module 1228 and/or remote data repository 1232, possibly for passage to the display 1208 after such processing or retrieval. The local processing and data module 1224 may be operatively coupled to the remote processing module 1228 and remote data repository 1232 by communication links 1236, 1240, such as via a wired or wireless communication links, such that these remote modules 1228, 1232 are operatively coupled to each other and available as resources to the local processing and data module 1224. The image capture device(s) can be used to capture the eye images used in the eye image segmentation, or eye tracking procedures.

In some embodiments, the remote processing module 1228 may comprise one or more processors configured to analyze and process data and/or image information such as video information captured by an image capture device (e.g., by performing RoomNet). The video data may be stored locally in the local processing and data module 1224 and/or in the remote data repository 1232. In some embodiments, the remote data repository 1232 may comprise a digital data storage facility, which may be available through the internet or other networking configuration in a “cloud” resource configuration. In some embodiments, all data is stored and all computations are performed in the local processing and data module 1224, allowing fully autonomous use from a remote module.

In some implementations, the local processing and data module 1224 and/or the remote processing module 1228 are programmed to perform embodiments of RoomNet to determine room layout. For example, the local processing and data module 1224 and/or the remote processing module 1228 can be programmed to perform embodiments of the process 1100 described with reference to FIG. 11. The local processing and data module 1224 and/or the remote processing module 1228 can be programmed to perform the room layout estimation method 1100 disclosed herein. The image capture device can capture video for a particular application (e.g., augmented reality (AR) or mixed reality (MR), human-computer interaction (HCl), autonomous vehicles, drones, or robotics in general). The video (or one or more frames from the video) can be analyzed using an embodiment of the computational RoomNet architecture by one or both of the processing modules 1224, 1228. In some cases, off-loading at least some of the RoomNet analysis to a remote processing module (e.g., in the “cloud”) may improve efficiency or speed of the computations. The parameters of the RoomNet neural network (e.g., weights, bias terms, subsampling factors for pooling layers, number and size of kernels in different layers, number of feature maps, room layout types, keypoint heat maps, etc.) can be stored in data modules 1224 and/or 1232.

The results of the RoomNet analysis (e.g., the output of the method 1100) can be used by one or both of the processing modules 1224, 1228 for additional operations or processing. For example, the processing modules 1224, 1228 of the wearable display system 1200 can be programmed to perform additional applications, such as augmented or mixed reality, indoor navigation, or scene reconstruction or rendering, based on the output of the method 1100. Accordingly, the wearable system 200 can use RoomNet to provide room layouts in real-time.

For example, the wearable display system 1200 can utilize a world map (e.g., stored in the local or remote data repositories 1224, 1240) that describes where objects, walls, floors, ceiling, doors, etc. are located relative to each other in a mixed reality environment. Further details regarding use of the world map are described in U.S. Patent Publication No. 2015/0016777, which is hereby incorporated by reference herein for all it contains. The output of RoomNet (e.g., the output of the method 1100) can be used to update the world map to include a room layout for a room in which the wearer of the system 1200 is located.

The RoomNet architecture can be used with other object recognizers or deep learning systems that analyze images for objects in the user's environment. For example, U.S. patent application Ser. No. 15/812,928, filed Nov. 14, 2017, entitled Deep Learning System for Cuboid Detection, which is hereby incorporated by reference herein in its entirety for all it contains, describes machine learning techniques to detect 3D cuboid-shaped objects in images. In some embodiments, the RoomNet architecture can be used to identify the room layout and the cuboid-detection architecture can be used to identify or localize cuboidal objects within the room layout. This information can be added to the world map of the wearable display system 1200 to provide an improved AR or MR user experience.

As yet another example, a robot can utilize embodiments of RoomNet to determine a room layout and then use the room layout for automated navigation of the robot within the room. Robots can include autonomous indoor robots (e.g., robotic vacuum cleaners, mops, sweepers), warehouse robots (e.g., used for automatic storage, retrieval, and inventory operations), indoor aerial drones, etc.

Additional Aspects

In a 1st aspect, a system for estimating a layout of a room is disclosed. the system comprises: non-transitory memory configured to store: a room image for room layout estimation; and a neural network for estimating a layout of a room, the neural network comprising: an encoder-decoder sub-network; and a classifier sub-network connected to the encoder-decoder sub-network; a hardware processor in communication with the non-transitory memory, the hardware processor programmed to: access the room image; determine, using the encoder-decoder sub-network and the room image, a plurality of predicted two-dimensional (2D) keypoint maps corresponding to a plurality of room types; determine, using the encoder-decoder sub-network, the classifier sub-network and the room image, a predicted room type from the plurality of room types; determine, using the plurality of predicted 2D keypoint maps and the predicted room type, a plurality of ordered keypoints associated with the predicted room type; and determine, using the plurality of ordered keypoints, a predicted layout of the room in the room image.

In a 2nd aspect, the system of aspect 1, wherein each room type in the plurality of room types comprises an ordered set of room type keypoints.

In a 3rd aspect, the system of aspect 2, wherein each room type in the plurality of room types comprises a semantic segmentation for regions in the room type, the semantic segmentation comprising an identification as a floor, a ceiling, or a wall.

In a 4th aspect, the system of aspect 2 or aspect 3, wherein a first keypoint order is associated with a first room type of the plurality of room types and a second keypoint order is associated with a second room type of the plurality of room types, wherein the first keypoint order and the second keypoint order are different.

In a 5th aspect, the system of any one of aspects 1 to 4, wherein the room image comprises a monocular image.

In a 6th aspect, the system of any one of aspects 1 to 5, wherein the room image comprises a Red-Green-Blue (RGB) image.

In a 7th aspect, the system of any one of aspects 1 to 6, wherein a dimensionality of the room image is larger than a dimensionality of the predicted 2D keypoint maps.

In an 8th aspect, the system of any one of aspects 1 to 7, wherein the encoder-decoder sub-network comprises an encoder sub-network comprising a plurality of convolutional layers and a plurality of pooling layers.

In a 9th aspect, the system of any one of aspects 1 to 8, wherein the encoder-decoder sub-network comprises a decoder sub-network comprising a plurality of convolutional layers and a plurality of upsampling layers.

In a 10th aspect, the system of any one of aspects 1 to 9, wherein the encoder-decoder sub-network comprises a memory augmented recurrent encoder-decoder (MRED) network.

In an 11th aspect, the system of any one of aspects 1 to 10, wherein the encoder-decoder sub-network comprises a plurality of recurrent layers.

In a 12th aspect, the system of aspect 11, wherein a number of recurrent iterations of the plurality of recurrent layers is two.

In a 13th aspect, the system of aspect 11, wherein a number of recurrent iterations of the plurality of recurrent layers is at least three.

In a 14th aspect, the system of any one of aspects 11 to 13, wherein each of the plurality of recurrent layers has a weight matrix, and the weight matrix is the same for all of the plurality of recurrent layers.

In a 15th aspect, the system of any one of aspects 1 to 14, wherein the predicted two-dimensional (2D) keypoint maps comprise heat maps.

In a 16th aspect, the system of aspect 15, wherein the hardware processor is programmed to extract keypoint locations from the heat maps as maxima of the heat map.

In a 17th aspect, the system of any one of aspects 1 to 16, wherein the hardware processor is programmed to: access object information from an object recognizer that analyzes the room image; and combine the object information with the predicted layout of the room.

In an 18th aspect, the system of aspect 17, wherein the object recognizer is configured to detect cuboids in the room image.

In a 19th aspect, a wearable display device is disclosed. The wearable display device comprises: an outward-facing imaging system configured to capture the room image for room layout estimation; and the system of any one of aspects 1 to 18.

In a 20th aspect, a system for training a neural network for estimating a layout of a room. The system comprises: non-transitory memory configured to store parameters for the neural network; and a hardware processor in communication with the non-transitory memory, the hardware processor programmed to: receive a training room image, wherein the training room image is associated with: a reference room type from a plurality of room types, and reference keypoints associated with a reference room layout; generate a neural network for room layout estimation, wherein the neural network comprises: an encoder-decoder sub-network configured to output predicted two-dimensional (2D) keypoints associated with a predicted room layout associated with each of the plurality of room types, and a side sub-network connected to the encoder-decoder network configured to output a predicted room type from the plurality of room types; and optimize a loss function based on a first loss for the predicted 2D keypoints and a second loss for the predicted room type; and update parameters of the neural network based on the optimized loss function.

In a 21st aspect, the system of aspect 20, wherein a number of the plurality of room types is greater than 5.

In a 22nd aspect, the system of aspect 20 or aspect 21, wherein the reference keypoints and the predicted 2D keypoints are associated with a keypoint order.

In a 23th aspect, the system of any one of aspects 20 to 22, wherein a first keypoint order is associated with a first room type of the plurality of room types and a second keypoint order is associated with a second room type of the plurality of room types, wherein the first keypoint order and the second keypoint order are different.

In a 24th aspect, the system of any one of aspects 20 to 23, wherein the training room image comprises a monocular image.

In a 25th aspect, the system of any one of aspects 20 to 24, wherein the training room image comprises a Red-Green-Blue (RGB) image.

In a 26th aspect, the system of any one of aspects 20 to 25, wherein a dimensionality of the training room image is larger than a dimensionality of a map associated with the predicted 2D keypoints.

In a 27th aspect, the system of any one of aspects 20 to 26, wherein the encoder sub-network and the decoder sub-network comprises a plurality of recurrent layers.

In a 28th aspect, the system of aspect 27, wherein a number of recurrent iterations of the recurrent layers is two or three.

In a 29th aspect, the system of aspect 27 or aspect 28, wherein deep supervision is applied to the recurrent layers.

In a 30th aspect, the system of any one of aspects 27 to 29, wherein weights associated with a first recurrent iteration of the iterations of the recurrent layers are identical to weights associated with a second recurrent iteration of the current layers.

In a 31st aspect, the system of any one of aspects 27 to 30, wherein the plurality of recurrent layers are configured as a memory-augmented recurrent encoder-decoder (MRED) network.

In a 32nd aspect, the system of any one of aspects 20 to 31, wherein the side sub-network comprises a room type classifier.

In a 33th aspect, the system of any one of aspects 20 to 32, wherein the first loss for the predicted 2D keypoints comprises a Euclidean loss between the plurality of reference keypoint locations and the predicted 2D keypoints.

In a 34th aspect, the system of any one of aspects 20 to 33, wherein the second loss for the predicted room type comprises a cross entropy loss.

In a 35th aspect, the system of any one of aspects 20 to 34, wherein the predicted 2D keypoints are extracted from a predicted heat map.

In a 36th aspect, the system of aspect 35, wherein hardware processor is programmed to: calculate a reference heat map associated with the reference keypoints of the training image; and calculate the first loss for the predicted 2D keypoints based on a difference between the predicted heat map and the reference heat map.

In a 37th aspect, the system of aspect 36, wherein the reference heat map comprises a two-dimensional distribution centered at a location for each reference keypoint.

In a 38th aspect, the system of aspect 36 or aspect 37, wherein the reference heat map comprises a background away from the reference keypoints and a foreground associated with the reference keypoints, and the hardware processor is programmed to weight gradients in the reference heat map based on a ratio between the foreground and the background.

In a 39th aspect, the system aspect 38, wherein to weight the gradients in the reference heat map, the hardware processor is programmed to degrade values of pixels in the background by a degrading factor less than one.

In a 40th aspect, a wearable display system is disclosed. The system comprises: an outward-facing imaging system configured to obtain a room image of an environment of the wearer of the wearable display system; non-transitory memory configured to store the room image; and a hardware processor in communication with the non-transitory memory, the processor programmed to: access the room image of the environment; and analyze the room image to determine a predicted layout of a room in the room image, wherein to analyze the room image, the processor is programmed to: use a neural network to determine an ordered set of two-dimensional (2D) keypoints associated with a room type for the room in the room image; and provide the room layout based at least partly on the 2D keypoints and the room type.

In a 41st aspect, the wearable display system of aspect 40, wherein the neural network comprises a convolutional encoder-decoder network.

In a 42nd aspect, the wearable display system of aspect 41, wherein the convolutional encoder-decoder network comprises a memory-augmented recurrent encoder-decoder network.

In a 43rd aspect, the wearable display system of any one of aspects 40 to 42, wherein the neural network comprises a classifier configured to determine the room type.

In a 44th aspect, the wearable display system of any one of aspects 40 to 43, wherein the hardware processor is further programmed to extract the ordered set of 2D keypoints from a heat map.

In a 45th aspect, the wearable display system of any one of aspects 40 to 44, wherein the hardware processor is further programmed to: access object information from an object recognizer that analyzes the room image; and combine the object information with the room layout.

In a 46th aspect, a method for estimating a layout of a room is disclosed. The method comprises: accessing a room image for room layout estimation; determining, using an encoder-decoder sub-network of a neural network for estimating a layout of a room and the room image, a plurality of predicted two-dimensional (2D) keypoint maps corresponding to a plurality of room types; determining, using the encoder-decoder sub-network, a classifier sub-network of the neural network connected to the encoder-decoder sub-network, and the room image, a predicted room type from the plurality of room types; determining, using the plurality of predicted 2D keypoint maps and the predicted room type, a plurality of ordered keypoints associated with the predicted room type; and determining, using the plurality of ordered keypoints, a predicted layout of the room in the room image.

In a 47th aspect, the method of aspect 46, wherein each room type in the plurality of room types comprises an ordered set of room type keypoints.

In a 48th aspect, the method of aspect 47, wherein each room type in the plurality of room types comprises a semantic segmentation for regions in the room type, the semantic segmentation comprising an identification as a floor, a ceiling, or a wall.

In a 49th aspect, the method of aspect 47 or aspect 48, wherein a first keypoint order is associated with a first room type of the plurality of room types and a second keypoint order is associated with a second room type of the plurality of room types, wherein the first keypoint order and the second keypoint order are different.

In a 50th aspect, the method of any one of aspects 46 to 49, wherein the room image comprises a monocular image.

In a 51st aspect, the method of any one of aspects 46 to 50, wherein the room image comprises a Red-Green-Blue (RGB) image.

In a 52nd aspect, the method of any one of aspects 46 to 51, wherein a dimensionality of the room image is larger than a dimensionality of the predicted 2D keypoint maps.

In a 53th aspect, the method of any one of aspects 46 to 52, wherein the encoder-decoder sub-network comprises an encoder sub-network comprising a plurality of convolutional layers and a plurality of pooling layers.

In a 54th aspect, the method of any one of aspects 46 to 53, wherein the encoder-decoder sub-network comprises a decoder sub-network comprising a plurality of convolutional layers and a plurality of upsampling layers.

In a 55th aspect, the method of any one of aspects 46 to 54, wherein the encoder-decoder sub-network comprises a memory augmented recurrent encoder-decoder (MRED) network.

In a 56th aspect, the method of any one of aspects 46 to 55, wherein the encoder-decoder sub-network comprises a plurality of recurrent layers.

In a 57th aspect, the method of aspect 56, wherein a number of recurrent iterations of the plurality of recurrent layers is two.

In a 58th aspect, the method of aspect 56, wherein a number of recurrent iterations of the plurality of recurrent layers is at least three.

In a 59th aspect, the method of any one of aspects 56 to 58, wherein each of the plurality of recurrent layers has a weight matrix, and the weight matrix is the same for all of the plurality of recurrent layers.

In a 60th aspect, the method of any one of aspects 46 to 59, wherein the predicted two-dimensional (2D) keypoint maps comprise heat maps.

In a 61th aspect, the method of aspect 60, further comprising extracting keypoint locations from the heat maps as maxima of the heat map.

In a 62th aspect, the method of any one of aspects 46 to 61, further comprising: accessing object information from an object recognizer that analyzes the room image; and combining the object information with the predicted layout of the room.

In a 63th aspect, the method of aspect 62, further comprising: using the object recognizer to detect cuboids in the room image.

In a 64th aspect, a method for training a neural network for estimating a layout of a room is disclosed. The method comprises: receiving a training room image, wherein the training room image is associated with: a reference room type from a plurality of room types, and reference keypoints associated with a reference room layout; generating a neural network for room layout estimation, wherein the neural network comprises: an encoder-decoder sub-network configured to output predicted two-dimensional (2D) keypoints associated with a predicted room layout associated with each of the plurality of room types, and a side sub-network connected to the encoder-decoder network configured to output a predicted room type from the plurality of room types; and optimizing a loss function based on a first loss for the predicted 2D keypoints and a second loss for the predicted room type; and updating parameters of the neural network based on the optimized loss function.

In a 65th aspect, the method of aspect 64, wherein a number of the plurality of room types is greater than 5.

In a 66th aspect, the method of aspect 64 or aspect 65, wherein the reference keypoints and the predicted 2D keypoints are associated with a keypoint order.

In a 67th aspect, the method of any one of aspects 64 to 66, wherein a first keypoint order is associated with a first room type of the plurality of room types and a second keypoint order is associated with a second room type of the plurality of room types, wherein the first keypoint order and the second keypoint order are different.

In a 68th aspect, the method of any one of aspects 64 to 67, wherein the training room image comprises a monocular image.

In a 69th aspect, the method of any one of aspects 64 to 68, wherein the training room image comprises a Red-Green-Blue (RGB) image.

In a 70th aspect, the method of any one of aspects 64 to 69, wherein a dimensionality of the training room image is larger than a dimensionality of a map associated with the predicted 2D keypoints.

In a 71th aspect, the method of any one of aspects 64 to 70, wherein the encoder sub-network and the decoder sub-network comprises a plurality of recurrent layers.

In a 72th aspect, the method of aspect 71, wherein a number of recurrent iterations of the recurrent layers is two or three.

In a 73th aspect, the method of aspect 71 or aspect 72, wherein deep supervision is applied to the recurrent layers.

In a 74th aspect, the method of any one of aspects 71 to 73, wherein weights associated with a first recurrent iteration of the iterations of the recurrent layers are identical to weights associated with a second recurrent iteration of the current layers.

In a 75th aspect, the method of any one of aspects 71 to 74, wherein the plurality of recurrent layers are configured as a memory-augmented recurrent encoder-decoder (MRED) network.

In a 76th aspect, the method of any one of aspects 64 to 75, wherein the side sub-network comprises a room type classifier.

In a 77th aspect, the method of any one of aspects 64 to 76, wherein the first loss for the predicted 2D keypoints comprises a Euclidean loss between the plurality of reference keypoint locations and the predicted 2D keypoints.

In a 78th aspect, the method of any one of aspects 64 to 77, wherein the second loss for the predicted room type comprises a cross entropy loss.

In a 79th aspect, the method of any one of aspects 64 to 78, wherein the predicted 2D keypoints are extracted from a predicted heat map.

In an 80th aspect, the method of aspect 79, further comprising: calculating a reference heat map associated with the reference keypoints of the training image; and calculating the first loss for the predicted 2D keypoints based on a difference between the predicted heat map and the reference heat map.

In an 81st aspect, the method of aspect 80, wherein the reference heat map comprises a two-dimensional distribution centered at a location for each reference keypoint.

In an 82th aspect, the method of aspect 80 or aspect 81, wherein the reference heat map comprises a background away from the reference keypoints and a foreground associated with the reference keypoints, and the hardware processor is programmed to weight gradients in the reference heat map based on a ratio between the foreground and the background.

In an 83th aspect, the method of aspect 82, wherein weighting the gradients in the reference heat map comprises degrading values of pixels in the background by a degrading factor less than one.

In an 84th aspect, a method is disclosed. The method comprises: accessing a room image of an environment; analyzing the room image to determine a predicted layout of a room in the room image, comprising: using a neural network to determine an ordered set of two-dimensional (2D) keypoints associated with a room type for the room in the room image; and providing the room layout based at least partly on the 2D keypoints and the room type.

In an 85th aspect, the method of aspect 84, wherein the neural network comprises a convolutional encoder-decoder network.

In an 86th aspect, the method of aspect 85, wherein the convolutional encoder-decoder network comprises a memory-augmented recurrent encoder-decoder network.

In an 87th aspect, the method of any one of aspects 84 to 86, wherein the neural network comprises a classifier configured to determine the room type.

In an 88th aspect, the method of any one of aspects 84 to 87, further comprising extracting the ordered set of 2D keypoints from a heat map.

In an 89th aspect, the method of any one of aspects 84 to 88, further comprising: analyzing the room image to determine object information; and combining the object information with the room layout.

Additional Considerations

Each of the processes, methods, and algorithms described herein and/or depicted in the attached figures may be embodied in, and fully or partially automated by, code modules executed by one or more physical computing systems, hardware computer processors, application-specific circuitry, and/or electronic hardware configured to execute specific and particular computer instructions. For example, computing systems can include general purpose computers (e.g., servers) programmed with specific computer instructions or special purpose computers, special purpose circuitry, and so forth. A code module may be compiled and linked into an executable program, installed in a dynamic link library, or may be written in an interpreted programming language. In some implementations, particular operations and methods may be performed by circuitry that is specific to a given function.

Further, certain implementations of the functionality of the present disclosure are sufficiently mathematically, computationally, or technically complex that application-specific hardware or one or more physical computing devices (utilizing appropriate specialized executable instructions) may be necessary to perform the functionality, for example, due to the volume or complexity of the calculations involved or to provide results substantially in real-time. For example, a video may include many frames, with each frame having millions of pixels, and specifically programmed computer hardware is necessary to process the video data to provide a desired image processing task (e.g., performance of the RoomNet techniques) or application in a commercially reasonable amount of time.

Code modules or any type of data may be stored on any type of non-transitory computer-readable medium, such as physical computer storage including hard drives, solid state memory, random access memory (RAM), read only memory (ROM), optical disc, volatile or non-volatile storage, combinations of the same and/or the like. The methods and modules (or data) may also be transmitted as generated data signals (e.g., as part of a carrier wave or other analog or digital propagated signal) on a variety of computer-readable transmission mediums, including wireless-based and wired/cable-based mediums, and may take a variety of forms (e.g., as part of a single or multiplexed analog signal, or as multiple discrete digital packets or frames). The results of the disclosed processes or process steps may be stored, persistently or otherwise, in any type of non-transitory, tangible computer storage or may be communicated via a computer-readable transmission medium.

Any processes, blocks, states, steps, or functionalities in flow diagrams described herein and/or depicted in the attached figures should be understood as potentially representing code modules, segments, or portions of code which include one or more executable instructions for implementing specific functions (e.g., logical or arithmetical) or steps in the process. The various processes, blocks, states, steps, or functionalities can be combined, rearranged, added to, deleted from, modified, or otherwise changed from the illustrative examples provided herein. In some embodiments, additional or different computing systems or code modules may perform some or all of the functionalities described herein. The methods and processes described herein are also not limited to any particular sequence, and the blocks, steps, or states relating thereto can be performed in other sequences that are appropriate, for example, in serial, in parallel, or in some other manner. Tasks or events may be added to or removed from the disclosed example embodiments. Moreover, the separation of various system components in the implementations described herein is for illustrative purposes and should not be understood as requiring such separation in all implementations. It should be understood that the described program components, methods, and systems can generally be integrated together in a single computer product or packaged into multiple computer products. Many implementation variations are possible.

The processes, methods, and systems may be implemented in a network (or distributed) computing environment. Network environments include enterprise-wide computer networks, intranets, local area networks (LAN), wide area networks (WAN), personal area networks (PAN), cloud computing networks, crowd-sourced computing networks, the Internet, and the World Wide Web. The network may be a wired or a wireless network or any other type of communication network.

The systems and methods of the disclosure each have several innovative aspects, no single one of which is solely responsible or required for the desirable attributes disclosed herein. The various features and processes described herein may be used independently of one another, or may be combined in various ways. All possible combinations and subcombinations are intended to fall within the scope of this disclosure. Various modifications to the implementations described in this disclosure may be readily apparent to those skilled in the art, and the generic principles defined herein may be applied to other implementations without departing from the spirit or scope of this disclosure. Thus, the claims are not intended to be limited to the implementations shown herein, but are to be accorded the widest scope consistent with this disclosure, the principles and the novel features disclosed herein.

Certain features that are described in this specification in the context of separate implementations also can be implemented in combination in a single implementation. Conversely, various features that are described in the context of a single implementation also can be implemented in multiple implementations separately or in any suitable subcombination. Moreover, although features may be described above as acting in certain combinations and even initially claimed as such, one or more features from a claimed combination can in some cases be excised from the combination, and the claimed combination may be directed to a subcombination or variation of a subcombination. No single feature or group of features is necessary or indispensable to each and every embodiment.

Conditional language used herein, such as, among others, “can,” “could,” “might,” “may,” “e.g.,” and the like, unless specifically stated otherwise, or otherwise understood within the context as used, is generally intended to convey that certain embodiments include, while other embodiments do not include, certain features, elements and/or steps. Thus, such conditional language is not generally intended to imply that features, elements and/or steps are in any way required for one or more embodiments or that one or more embodiments necessarily include logic for deciding, with or without author input or prompting, whether these features, elements and/or steps are included or are to be performed in any particular embodiment. The terms “comprising,” “including,” “having,” and the like are synonymous and are used inclusively, in an open-ended fashion, and do not exclude additional elements, features, acts, operations, and so forth. Also, the term “or” is used in its inclusive sense (and not in its exclusive sense) so that when used, for example, to connect a list of elements, the term “or” means one, some, or all of the elements in the list. In addition, the articles “a,” “an,” and “the” as used in this application and the appended claims are to be construed to mean “one or more” or “at least one” unless specified otherwise.

As used herein, a phrase referring to “at least one of” a list of items refers to any combination of those items, including single members. As an example, “at least one of: A, B, or C” is intended to cover: A, B, C, A and B, A and C, B and C, and A, B, and C. Conjunctive language such as the phrase “at least one of X, Y and Z,” unless specifically stated otherwise, is otherwise understood with the context as used in general to convey that an item, term, etc. may be at least one of X, Y or Z. Thus, such conjunctive language is not generally intended to imply that certain embodiments require at least one of X, at least one of Y and at least one of Z to each be present.

Similarly, while operations may be depicted in the drawings in a particular order, it is to be recognized that such operations need not be performed in the particular order shown or in sequential order, or that all illustrated operations be performed, to achieve desirable results. Further, the drawings may schematically depict one more example processes in the form of a flowchart. However, other operations that are not depicted can be incorporated in the example methods and processes that are schematically illustrated. For example, one or more additional operations can be performed before, after, simultaneously, or between any of the illustrated operations. Additionally, the operations may be rearranged or reordered in other implementations. In certain circumstances, multitasking and parallel processing may be advantageous. Moreover, the separation of various system components in the implementations described above should not be understood as requiring such separation in all implementations, and it should be understood that the described program components and systems can generally be integrated together in a single software product or packaged into multiple software products. Additionally, other implementations are within the scope of the following claims. In some cases, the actions recited in the claims can be performed in a different order and still achieve desirable results. 

What is claimed is:
 1. A system comprising: non-transitory memory configured to store: an image of a room; and a neural network comprising: an encoder-decoder sub-network comprising an encoder and a decoder; and a classifier sub-network in communication with the encoder-decoder sub-network; a hardware processor in communication with the non-transitory memory, the hardware processor programmed to: access the image; predict, using the encoder, decoder, and the image, two-dimensional (2D) keypoint maps corresponding to room types; determine, using the encoder, the classifier sub-network, and the image, a predicted room type from the room types; and determine a predicted layout of the room in the image based at least in part on the 2D keypoint maps and the predicted room type.
 2. The system of claim 1, wherein each room type in the room types comprises a semantic segmentation for regions in the room type, the semantic segmentation comprising an identification as a floor, a ceiling, or a wall.
 3. The system of claim 1, wherein a first keypoint order is associated with a first room type of the room types and a second keypoint order is associated with a second room type of the room types, wherein the first keypoint order and the second keypoint order are different.
 4. The system of claim 1, wherein the encoder comprises a plurality of convolutional layers and a plurality of pooling layers, or the decoder comprises a plurality of convolutional layers and a plurality of upsampling layers.
 5. The system of claim 1, wherein the encoder-decoder sub-network comprises a plurality of recurrent layers.
 6. The system of claim 5, wherein a number of recurrent iterations of the plurality of recurrent layers is at least two.
 7. The system of claim 1, wherein the predicted two-dimensional (2D) keypoint maps comprise heat maps.
 8. The system of claim 7, wherein the hardware processor is programmed to extract keypoint locations from the heat maps as maxima of the heat map.
 9. The system of claim 1, wherein the hardware processor is programmed to: access object information from an object recognizer that analyzes the image; and combine the object information with the predicted layout of the room.
 10. The system of claim 9, wherein the object recognizer is configured to detect cuboids in the image.
 11. A system comprising: non-transitory memory configured to store parameters for the neural network; and a hardware processor in communication with the non-transitory memory, the hardware processor programmed to: receive an image, wherein the image is associated with: a reference room type from a plurality of room types, and reference keypoints associated with a reference room layout; generate a neural network comprising: an encoder-decoder sub-network configured to output two-dimensional (2D) keypoints associated with a predicted room layout, wherein the encoder-decoder sub-network comprises an encoder and a decoder, and a side sub-network in communication with the encoder-decoder network configured to output a predicted room type from the plurality of room types; and update the parameters for the neural network based at least in part on an optimized loss function.
 12. The system of claim 11, wherein a first keypoint order is associated with a first room type of the plurality of room types and a second keypoint order is associated with a second room type of the plurality of room types, wherein the first keypoint order and the second keypoint order are different.
 13. The system of claim 11, wherein the encoder and the decoder comprise a plurality of recurrent layers.
 14. The system of claim 11, wherein the optimized loss function is based at least in part on a first loss for the 2D keypoints and a second loss for the predicted room type.
 15. The system of claim 11, wherein the 2D keypoints are extracted from a predicted heat map.
 16. The system of claim 15, wherein hardware processor is programmed to: calculate a reference heat map associated with the reference keypoints; and calculate the first loss for the 2D keypoints based on a difference between the predicted heat map and the reference heat map.
 17. A wearable display system comprising: an outward-facing imaging system configured to obtain an image of an environment of the wearer of the wearable display system; non-transitory memory configured to store the image; and a hardware processor in communication with the non-transitory memory, the processor programmed to: access the image; and analyze the image to determine a predicted layout of a room in the image, wherein to analyze the image, the processor is programmed to: use a neural network to determine two-dimensional (2D) keypoints associated with a room type associated with the image, wherein the neural network comprises: an encoder-decoder sub-network configured to output the 2D keypoints, and a side sub-network in communication with the encoder-decoder sub-network configured to output the room type; and provide the room layout based at least partly on the 2D keypoints and the room type.
 18. The wearable display system of claim 17, wherein the neural network comprises a convolutional encoder-decoder network.
 19. The wearable display system of claim 17, wherein the neural network comprises a classifier configured to determine the room type.
 20. A method comprising: accessing an image of a room; predicting two-dimensional (2D) keypoint maps corresponding to room types based at least in part on the image; determining a predicted room type from the room types based at least in part on the image; and determining a predicted layout of the room in the image based at least in part on the 2D keypoint maps and the predicted room type.
 21. The method of claim 20, wherein predicting 2D keypoint maps is also based at least in part on a neural network.
 22. The method of claim 21, wherein determining a predicted room type is also based at least in part on a neural network.
 23. The method of claim 20, wherein each room type in the room types comprises a semantic segmentation for regions in the room type, the semantic segmentation comprising an identification as a floor, a ceiling, or a wall.
 24. The method of claim 20, wherein a first keypoint order is associated with a first room type of the room types and a second keypoint order is associated with a second room type of the room types, wherein the first keypoint order and the second keypoint order are different. 